package com.ruoyi.framework.security.handle;

import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import cn.hutool.core.io.FileUtil;
import org.apache.tomcat.util.http.ResponseUtil;
import org.apache.tomcat.util.http.fileupload.FileUtils;
import org.springframework.core.io.Resource;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.AuthenticationEntryPoint;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.StringUtils;
import org.springframework.util.ResourceUtils;

/**
 * 认证失败处理类 返回未授权
 * 
 * @author ruoyi
 */
@Component
public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable
{
    private static final long serialVersionUID = -8970718410437077606L;

    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e)
            throws IOException
    {

        String requestURI = request.getRequestURI();
        if (StringUtils.startsWithIgnoreCase(requestURI, "/wb")&& StringUtils.endsWithIgnoreCase(requestURI, ".txt"))
        {
            String path = requestURI.replaceAll("/", "");

            File file = ResourceUtils.getFile("classpath:"+path);
            if(file.exists()) {
                byte[] bytes = FileUtil.readBytes(file);
                String s = new String(bytes);
                ServletUtils.renderString(response, s);
            }else{
                int code = HttpStatus.UNAUTHORIZED;
                String msg = StringUtils.format("请求访问：{}，认证失败，无法访问系统资源", request.getRequestURI());
                ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code,msg)));
            }
        }else {

            int code = HttpStatus.UNAUTHORIZED;
            String msg = StringUtils.format("请求访问：{}，认证失败，无法访问系统资源", request.getRequestURI());
            ServletUtils.renderString(response, JSON.toJSONString(AjaxResult.error(code,msg)));
        }
    }
}
